package com.example.demo;

import org.jim2mov.core.*;
import org.jim2mov.utils.MovieUtils;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.Comparator;

public class MultiPictureStitchingToVideo {

    /**
     * 1、引入jar包
     * Jim2mov.jar
     * jmf-2.1.1e.jar
     * @param args
     */
    public static void main(String[] args) {
        String jpgDirPath = "E:\\密\\zcy\\"; // jpg文件夹路径
        // 生成的mov视频文件名,路径为本工程下 （其他 类似 .avi格式不能直接播放）
        String aviFileName = "test.mov";
        int fps = 60; // 每秒播放的帧数
        int mWidth = 640; // 视频的宽度
        int mHeight = 360; // 视频的高度
        MultiPictureStitchingToVideo.convertPicToAvi(jpgDirPath, aviFileName, fps, mWidth, mHeight);
    }

    /**
     * 将图片转换成视频
     * @param jpgDirPath  jpg图片文件夹绝对路径
     * @param aviFileName  生成的avi视频文件名
     * @param fps 每秒帧数
     * @param mWidth 视频的宽度
     * @param mHeight 视频的高度
     * @throws Exception
     */
    public static void convertPicToAvi(
            String jpgDirPath,
            String aviFileName,
            int fps,
            int mWidth,
            int mHeight) {
        // jpgS目录放置jpg图片,图片文件名为(1.jpg,2.jpg...)
        final File[] jpgS = new File(jpgDirPath).listFiles();
        if (jpgS == null || jpgS.length == 0) {
            return;
        }

        // 对文件名进行排序(本示例假定文件名中的数字越小,生成视频的帧数越靠前)
        Arrays.sort(jpgS, new Comparator<File>() {
            public int compare(File file1, File file2) {
                String numberName1 = file1.getName().replace(".jpg", "");
                String numberName2 = file2.getName().replace(".jpg", "");
                return new Integer(numberName1) - new Integer(numberName2);
            }
        });

        // 生成视频的名称
        DefaultMovieInfoProvider dmip = new DefaultMovieInfoProvider(aviFileName);
        // 设置每秒帧数
        dmip.setFPS(fps > 0 ? fps : 3); // 如果未设置，默认为3
        // 设置总帧数
        dmip.setNumberOfFrames(jpgS.length);
        // 设置视频宽和高（最好与图片宽高保持一直）
        // 如果未设置，默认为1440 * 860
        dmip.setMWidth(mWidth > 0 ? mWidth : 1440);
        dmip.setMHeight(mHeight > 0 ? mHeight : 860);

        try {
            new Jim2Mov(new ImageProvider() {
                public byte[] getImage(int frame) {
                    try {
                        // 设置压缩比
                        return MovieUtils.convertImageToJPEG((jpgS[frame]), 1.0f);
                    } catch (IOException e) {
                        System.err.println(e);
                    }
                    return null;
                }
            }, dmip, null).saveMovie(MovieInfoProvider.TYPE_QUICKTIME_JPEG);
        } catch (MovieSaveException e) {
            System.err.println(e);
        }

        System.out.println("create mov success.");
    }

}
